home *** CD-ROM | disk | FTP | other *** search
/ MacHack 2000 / MacHack 2000.toast / pc / The Hacks / MacHacksBug / Python 1.5.2c1 / Mac / Lib / test / tmenu.py < prev    next >
Encoding:
Python Source  |  2000-06-23  |  1.4 KB  |  69 lines
  1. # Create hierarchical menus for some volumes.
  2.  
  3. import os
  4. from Menu import *
  5. import macfs
  6. import sys
  7.  
  8. def main():
  9.     global oldbar
  10.     my_volumes = []
  11.     while 1:
  12.         fss, ok = macfs.GetDirectory()
  13.         if not ok:
  14.             break
  15.         my_volumes.append(fss.as_pathname())
  16.     if not my_volumes:
  17.         return
  18.     oldbar = GetMenuBar()
  19.     ClearMenuBar()
  20.     makevolmenus(my_volumes)
  21.     DrawMenuBar()
  22.  
  23. def reset():
  24.     oldbar.SetMenuBar()
  25.     DrawMenuBar()
  26.  
  27. id = 1
  28. def nextid():
  29.     global id
  30.     nid = id
  31.     id = id+1
  32.     return nid
  33.  
  34. def makevolmenus(volumes):
  35.     for vol in volumes:
  36.         makevolmenu(vol)
  37.  
  38. def makevolmenu(vol):
  39.     menu = NewMenu(nextid(), vol)
  40.     adddirectory(menu, vol)
  41.     menu.InsertMenu(0)
  42.  
  43. def adddirectory(menu, dir, maxdepth = 1):
  44.     print "adddirectory:", `dir`, maxdepth
  45.     files = os.listdir(dir)
  46.     item = 0
  47.     for file in files:
  48.         item = item+1
  49.         menu.AppendMenu('x')        # add a dummy string
  50.         menu.SetMenuItemText(item, file)    # set the actual text
  51.         fullname = os.path.join(dir, file)
  52.         if os.path.isdir(fullname):
  53.             menu.SetMenuItemText(item, ':' + file + ':')    # append colons
  54.             if maxdepth > 0:
  55.                 id = nextid()
  56.                 submenu = NewMenu(id, fullname)
  57.                 adddirectory(submenu, fullname, maxdepth-1)
  58.                 submenu.InsertMenu(-1)
  59.                 # If the 'Cmd' is 0x1B, then the 'Mark' is the submenu id
  60.                 menu.SetItemMark(item, id)
  61.                 menu.SetItemCmd(item, 0x1B)
  62.     if not files:
  63.         menu.AppendMenu(':')    # dummy item to make it selectable
  64.     return menu
  65.  
  66. if __name__ == '__main__':
  67.     main()
  68.     sys.exit(1)   # To allow the user to interact...
  69.